import os

import yaml

def write_yaml(data):
    with open(os.getcwd()+"./extract.yaml", encoding="utf-8", mode="a+") as f:
        yaml.dump(data, stream=f, allow_unicode=True)

def read_yaml():
    with open(os.getcwd()+"./extract.yaml", encoding="utf-8", mode="r") as f:
        value = yaml.load(f, yaml.FullLoader)
        return value

def clear_yaml():
    with open(os.getcwd()+"./extract", encoding="utf-8", mode="w") as f:
        f.truncate()

# 读取测试用例
def read_testcase_yaml(yamlpath):
    with open(os.getcwd()+"/"+yamlpath, encoding="utf-8", mode="r") as f:
        value = yaml.load(f, yaml.FullLoader)
        return value